from .q_learner import QLearner
from .iq_learner import IQLearner
from .gail_learner import GAILLearner


REGISTRY = {}
REGISTRY["q_learner"] = QLearner
REGISTRY["iq_learner"] = IQLearner
REGISTRY["gail_learner"] = GAILLearner
